homework01 final version - no multi classify