Added simple functionality tests