Commented code. Updated readme