Reworked script with new class