scala: week4 getting close