minor fix
added stage picker