updates code cleanup