Initial solution for ex 1-5