Added some registration code