Alter lay snippet to use extends