Modified resgister and added a functions.php