Added integers back
Added memoization and input
Updated TODO