Updated due to API changes.

Signed-off-by: Man_cub <mcmancub@gmail.com>
Merge pull request #2 from MCManCub/master

Updated due to API changes.