Fixed error. still learning...