Added answer and cleaned up comment