Adjusted layout in function.php