Completed assignment

minor change to add =matrix() to function