How to convert this javascript to function to work on button click