Update functions.php

Fixed return types