Added some more documentation

trying to make a little bit better code