added function to color picture un yellow
removed unused import