Updated functions.php to only return requested data