Function works, testing added