finally got the js to work but needs styling