finish code. maybe tessting, for Lesson3