Partial addition of non-linear elements and multipoles to initialize_element subroutine