Make code grant flow work