Updated Source Code from today's lecture