UI changes, better mouse control