Slight modifications to lego main