Update functions.php

Added custom functionality