Small changes to how the code connects/outputs