Adding some color and making the animation look better