Fixed arrowhead code structure

Solution to leveluprails arrowhead exercise