Modified code to fit phone screen