Adding comments and small stuff

Adding comments, example and updating TODO