Adding colors and a shutdown hook