small update + comments
almost done session 4