cleanup python
added back in python code