Working on implementing stage 3 of fast aks