OOP Python Classes completion for level 1