Minor Update to fix errors