Add remove function

Add function to remove a row