Cleaned up all files according too PEP8