JT-labFour modified on code according to comments