Minor changes

Added loop, changed color scheme