demo gui plus better level prediction
code refactor