Started added logic to the functions.php file.