Correction

Added things that got removed by accident