fixed a couple of typos while using in a bio 599 lecture; reduced runtime slightly
fixed missing import