finishing touches on code