Just fixed the comments

Ready for testing