style of the code was fixed