Refactored the Element classes to return div to be added to main div