Fix syntax errors

Yes, I'm a noob at python.. :)