Make code grant flow work as well