my code for gschool warmup problems.