cleaned up the code. Also working on a dropdown that contains the update and remove functions