small python exam