added mit license + small tidy up