Updates methods to make Myles' driver code work