Cleaned code

1 bug