Added a remove code, and quickly wrote the logic for removing a file