Added custom registration controller back