fix for lecture 1