Sample code for boss stage 3