Add remove function to controller
Add find function to controller