Edited some TODO:s