Add more layout to Reader Register Training