Added methods to add and remove button