Update functions.php

Fix current page