Added "add assignment" functionality