Small changes, got rid of console log statement.