Updated Repo

Added visualization and color scheme, fixed broken features