Feature/add js and button