Made class files work with Unity