Fixed some styling class issues