Small fixes, added documentation