Final version. Menu flow works