Minor updates to registration