Add colors and adjust ui for each screen