Add some basic JS logic