Refactor wheel into a div