Add ability to add classes in to screen