Added cool things to the code