First version of week6 assignement