Use built-in filter() function for speed

Also it's way more obvious.