fixed some error in code

Signed-off-by: Nicholas Tan Jerome <nicholas.jerome@kit.edu>