updated challenge3 to use pyrax