Main Fix

fixed main