Signed-off-by: Yulong Bai <yulongb@stud.ntnu.no>