fixing more code
fixing more code