Adds if statement to test, adds comments