Add bool type check for if and while stmts - check #4