Updated code

Made necessary changes for everyone to work with their part.