Interactive Python: Final assignment (RiceRocks)