‘Drawing’ the wrong conclusions

The teacher keeps telling my son to stop drawing in class, but he says he focuses better when he does. Should I pursue a 504 plan so he can continue?