Revised assignment 4.1

Included a "return array;" and removed .filter per Martin's comments