Adding a proper button