Addition of non-linear elements to the initialize_element subroutine