Changed from button to div