Updates to clean up code
Ignore .py~