Just figuring out classes in python. Minimal changes