Write more documentation