added test files + adding doc strings