Updating with styling
Updating code div bits