Machine learning course assignment 4.