Add more comments

Add comments + make them more understable