Minor fixes / added comments for tests