Fixed the stub in the main.. Almost forgot