Crappy first python script