Add logic for changing arrow