My beginner python code