"add class" functionality and layout changes