Color coding done, now just need to resize