Fixed bug setting image width when I shouldn't have, nicer color dispalay UI