Add calculator code