Code optimisation + animation

Few code cleaning. Change the color output and added an infinite
drawing loop (setInterval)