cleaned up code, made more readable, modified plotting colors