Added code for responsive screen size