Started to add remove document function