Add warning message when screen is too small
Remove heading test
Remove extra padding