Updates to test and implement the add_mapping method