Double post or related to this post. Are you working on the same problem and where are you stuck implementing the red part?