Cleaned up test code
fixed newline problem