Added more styling using buttons