THIS IS HOW IT IS SUPPOSED TO BE AT THE END