My code for the peer assignment, version 2