Add functional to Register.php