Added div classes back