Starting to make a colorbar function