Not quite done, but adding assignment input