functionality works but UI needs to be fixed