Added bits of functionality + legend to UI