added legends, future work, removed typos