Fixing main script in package