Adding screen size in