Add layout functionality