Add logic in comments