Add remove file functionality