Added screen size check and added more styling