SOme minor correction (to make it run :p)