Add first part of hw solution