slight changes to design