Tidy up lambda