Removed Caffinate call and did a bit of pep8 sprucing