Updated checking/Added comments

Should be better.