Adding 2nd part of hw