added restaraunt class and button/text on main screen