fixed up ui