finished up UI work