Changes to make Add button work.