Fixed bug. All left to do now is create a delete button.