minor changes for ass1 + added NOTES