cleaned up some code after pycheckig