Almost there, just need add console and clean/comment code