Improve styling on mobile screen and hide virtual screen when button clicked