Added color logic to heatmap image function

Color function sourced from https://github.com/jeffkaufman/apartment_prices instead of duplicating work.