Cleanup heat.py

Removed un-needed code from the host lookup, and cleaned up the code
for pep8 errors.