minor changes to the design