Added inheritance class structure to check different parts of the system