Almost done with logic, need to resolve one error.