Got rid of unnecessary button element