added code from today's class