refactoring code snippets in python