Test cases left.

Write test cases.

Check if properly commented again.