Cleanup

Some documentation too.