moved code snippet where its suppose to go, kinda