Made some slight amendments to the design