bugfixes

I'm just learning Python! :D