Updated the reference to package main