Update sample to use pub package layout