solve code snippets in python