Adding class after load feature