added stuff in functies.php