python refactoring tips