Updated code to match that of Prototype 0.2 (Fritzing)

Updated comments as well to make easier to understand