Added class for new screens IOS