Python code, meant to practice use of classes