removed the stage ( it was useless ) probably other things, yolo