addition of colors and the ability to run with command line arguments