add comments
connect lego fix #4