Updated UI

TODO: Outputs should be images rather than text.