Minor plot alterations