Modify

Signed-off-by: Morteza Kheirkhah <m.kheirkhah@sussex.ac.uk>