add comment + error fix