Implementing into main and fixing problem with enemies