Added class assignment on load