Proposing lines 4-5

Please add lines 4-5
@damithc