week4 HW Multi-class Classification and Neural Networks finished