fixed with correct inputs arguments