edited UI, I might make more changes if I have time