correcting in lecture 2
some improements