fix display on mobile

Fix #2 and #3