Avnit : clean up code