tidying, fixed an error!