fixed the form of the code