Add `current_branch` function