Add code from lecture
Add code after lecture