updating the code to look normal