correct sol for q01