Machine learning course work