Trying to get better at programming in python.