Chnaged source code agian following tutorial!