Vik/add docs

Add in more documentation.