added better code noob