Added rainbow feature and arrow key controls