Refactor AV code