Added colors and save states to ui.py. Updated README.