added colors and fixed some pep8