touch up to ui - DONT MODIFY