Add more tests and refactor wiki_page.py