added comments to remind me where I need to make improvements