Code seems to work now