Removed example code and small fix