Fixed code, it now runs