Add colormaps, basic UI, make it look a little pretty