some changes after solution video