Added idea

made a note to try creating class detection for the future.