Finish up code