Added an edge-case test