Remove whitespaces and update TODO