Made small change in architecture