Fix/remove TODOs

Updates #14