Update functions.php

Minor fixes