Fix to avoid warning message taking address of main

Fixes #82
Takes just another function instead of main. This way we keep people using the standard happy.