Fixed some code errors!