adding layout back because i realized it is necessary