Adding action for buttons and first attempt to fix mobile version