Fix the PlusOne exercise's python solution