Documentation / comments added