Added Tests

Added nice tests to ensure proper functionality.