made minor changes on lecture 3