Added some JS functionality :)