easy problem, but not implement it in a easy way