Adding classes and edit profile button