Small update to sample's main