Trying python code