Almost have mutate working