added the updated main