AI Course assignment solution