Added rudimentary colorbar
Added some colorbar stuff