Added custom JS functionality