Add layout tests