Functionality now in place. Just need to refactor and maybe add in confirm messages.