Updations in register

Fix bugs