clean code + add todos