Updated for fixing div class