Solution for week2