And alwyays use if __name__ == '__main__' if you want be able to test your code.