updated classes for styling