Add a simple filtering function
Add tests for our filter function
More test cases