Add classes for styling