Write a version using pyrax