Added flexbox to home screen