Adding mock screen flow