fixed code to match style of repository