cleaning up, adding doc