Cleaned up and added comments
First steps of docs