Exercise1, first point
The second step took me really a while, I hope the solution is not overcomplicated
Added the remove method