Add references to related work, define probability space