Converted test into actually checking things (not printing results only)