Add sample code for boss stage 1