Working on 2D model