Minor bug fix - Add button