Changed display from class to function. This is to make customizing easier

Status:Working but not styled.