Add callback for code execution on animation complete