Added divs to make a better user experience