Changed code to what we did in class on Tuesday (2/4/14).