made some slight style changes