problem-solving code snippets in python