Added plus button and made it functional