Adding mycodeschool problem of how to return an array