Finished work on issue #4, adding deletion function