adding last touch of code