fixed issue 5 by adding