Updated AK model