Add logic in app